op {
  name: "ExtractJpegShape"
  input_arg {
    name: "contents"
    type: DT_STRING
  }
  output_arg {
    name: "image_shape"
    type_attr: "output_type"
  }
  attr {
    name: "output_type"
    type: "type"
    default_value {
      type: DT_INT32
    }
    allowed_values {
      list {
        type: DT_INT32
        type: DT_INT64
      }
    }
  }
}
